# test03_dict.py
# 模拟用户注册
# 使用字典保存用户名和密码
# 创建函数register完成用户注册，要求用户先输入用户名，判断是否存在该用户，如果存在提示‘该用户已被注册’，否则不存在提示用户输入密码，将用户名和密码保存到字典中，并提示注册时成功
users = {'zs': '123456'}
def register():
    username = input('请输入用户名：')
    if username in users:
        print('该用户已被注册')
    else:
        password = input('请输入密码：')
        users[username] = password
        print('注册成功！', users)
# register()
# 练习：创建函数login，完成登录功能
# 要求用户输入用户名和密码，判断字典中是否存在该用户，如果存在打印‘登录成功’，否则登录失败
def login():
    username = input('请输入用户名:')
    password = input('请输入密码:')
    # 将用户名和密码放入到一个元组，然后再判断是否存在
    if (username, password) in users.items():
        print('登录成功')
    else:
        print('登录失败')
# login()
def show_menu():
    while True:
        n = input('1-注册 2-登录 3-退出 请输入操作：')
        if n not in ['1', '2', '3']:
            print('错误的操作！')
        elif n == '1':
            register()
        elif n == '2':
            login()
        else:
            print('bye~')
            break
show_menu()